import os
import glob
import sys

# 设置目标目录路径（保留原始路径中的空格）
target_dir = r"D:\魔方\oll_dys"

# 检查目录是否存在
if not os.path.exists(target_dir):
    print(f"错误：目录不存在 - {target_dir}")
    sys.exit(1)
if not os.path.isdir(target_dir):
    print(f"错误：{target_dir} 不是有效的目录")
    sys.exit(1)

# 支持的图片文件扩展名（不区分大小写）
image_extensions = ('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff', '.webp')

# 获取目录下所有图片文件
image_files = []
for ext in image_extensions:
    # 匹配小写扩展名
    image_files.extend(glob.glob(os.path.join(target_dir, f'*{ext}')))
    # 匹配大写扩展名
    image_files.extend(glob.glob(os.path.join(target_dir, f'*{ext.upper()}')))

# 去重处理（防止同一文件被多次添加）
image_files = list(set(image_files))

# 按文件创建时间升序排序
image_files.sort(key=lambda x: os.path.getctime(x))

# 重命名文件
for index, old_path in enumerate(image_files, start=1):
    # 获取文件扩展名
    file_ext = os.path.splitext(old_path)[1].lower()
    # 生成新文件名（两位数序号 + 原始扩展名）
    new_filename = f"{index:02d}{file_ext}"
    new_path = os.path.join(target_dir, new_filename)
    
    # 执行重命名
    os.rename(old_path, new_path)
    print(f"重命名成功: {os.path.basename(old_path)} -> {new_filename}")

print("所有图片文件已按时间升序重命名完成！")